#import  "AppBundle.h"
#import "WKApp.h"
NSBundle * _Nonnull getAppBundle() {
    NSBundle *bundle = [NSBundle mainBundle];
    if ([[bundle.bundleURL pathExtension] isEqualToString:@"appex"]) {
        bundle = [NSBundle bundleWithURL:[[bundle.bundleURL URLByDeletingLastPathComponent] URLByDeletingLastPathComponent]];
    } else if ([[bundle.bundleURL pathExtension] isEqualToString:@"framework"]) {
        bundle = [NSBundle bundleWithURL:[[bundle.bundleURL URLByDeletingLastPathComponent] URLByDeletingLastPathComponent]];
    } else if ([[bundle.bundleURL pathExtension] isEqualToString:@"Frameworks"]) {
        bundle = [NSBundle bundleWithURL:[bundle.bundleURL URLByDeletingLastPathComponent]];
    }
    return bundle;
}

@implementation UIImage (AppBundle)

- (instancetype _Nullable)initWithBundleImageName:(NSString * _Nonnull)bundleImageName {
//    return [UIImage imageNamed:bundleImageName inBundle:getAppBundle() compatibleWithTraitCollection:nil];
    return [WKApp.shared loadImage:bundleImageName moduleID:@"WuKongBase"];
}

@end
